package com.yaorange.common.utils.nongjt;

import java.util.Random;

public class ValidateCodeUtils {
    /**
     * 随机⽣成验证码
     *
     * @param length ⻓度为4位或者6位
     * @return
     */
    public static String generateValidateCode(int length) {
        Integer code = null;
        if (length == 4) {
            // ⽣成随机数，最⼤为9999
            code = new Random().nextInt(9999);
            if (code < 1000) {
                // 保证随机数为4位数字
                code = code + 1000;
            }
        } else if (length == 6) {
            // ⽣成随机数，最⼤为999999
            code = new Random().nextInt(999999);
            if (code < 100000) {
                // 保证随机数为6位数字
                code = code + 100000;
            }
        } else {
            throw new RuntimeException("只能⽣成4位或6位数字验证码");
        }
        return String.valueOf(code);
    }
    /**
     * 随机⽣成指定⻓度字符串验证码 带字⺟
     *
     * @param length ⻓度
     * @return
     */
    public static String generateValidateCodeString(int length) {
        Random rdm = new Random();
        String hash1 = Integer.toHexString(rdm.nextInt());
        String capstr = hash1.substring(0, length);
        return capstr;
    }
}
